pytorch卸载cuda+cudnn并重新配置GPU环境,亲测有效 |
您所在的位置:网站首页 › usb keyboard驱动彻底卸载 › pytorch卸载cuda+cudnn并重新配置GPU环境,亲测有效 |
pytorch卸载cuda+cudnn
一、卸载cuda
进入【控制面板】,点击【卸载程序】 将红色框中带版本号的都卸载 1、进入安装路径 将以下版本号文件直接删除 1、win+r,输入cmd,【回车】进入命令行窗口
cuda官网 1、选择对应版本的cuda进行下载(我选择下载了11.7.1版本的) 点进去后,下载在线的安装包,该安装包比较小 2、配置环境变量(默认是配置好了的,没有的自己加上) 右键【此电脑】→点击【属性】→点击【高级系统设置】→点击【环境变量】,在【系统变量】中 首先是这两个路径 3、测试cuda是否安装成功 在命令行输入nvcc -V,出现以下版本信息,即为成功 nvcc -Vcudnn官网 1、登录cudnn后,选择对应版本的进行下载。 我的cuda是11.7,因此下载最新windows版本的11.x的cudnn就行。 1、选择对应python版本 cuda对应python版本关系 可以发现,cuda11.7对应的pytorch、python等版本,因为我要用pytorch2.x,对应的python[3.8,3.11],我选择python3.10 创建后进入虚拟环境 conda activate pytorch_two1、查找对应版本 官网对应版本关系 Ctr+F搜索 需要安装的pytorch版本(2.0.1),找到windows下cuda版本(11.7)对应的torchvision版本(0.15.2)、torchaudio版本(2.0.2) 下载好的组件 在刚刚新建的anaconda虚拟环境中本地安装这3个组件。 依次运行以下3条命令 pip install F:\pytorch_whl\torch-2.0.1+cu117-cp310-cp310-win_amd64.whl pip install F:\pytorch_whl\torchaudio-2.0.2+cu117-cp310-cp310-win_amd64.whl pip install F:\pytorch_whl\torchvision-0.15.2+cu117-cp310-cp310-win_amd64.whl
|
今日新闻 |
推荐新闻 |
CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |